[HDU 2544] 最短路 · 堆优化dijkstra

模板题用来练手。

现在来说一般的图论题目都很难用普通dijkstra过掉，而SPFA又很不稳定，还是学了一下国际公认的堆优化dijkstra。

简单来说，堆优化dij就是把for循环找最小的d[i]那维用堆来做，将O(n)降成了O(logn)。

关于一个小问题见程序注释

【据说priority_queue常数巨大 ，不管了】

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;

const int N=10005;
int node[N],nxt[N],head[N],data[N];
int n,m,d[N],tot;
int i,j;

void add(int x,int y,int z){
    node[++tot]=y;nxt[tot]=head[x];head[x]=tot;data[tot]=z;
}

void init(){
    tot=0;
    memset(head,0,sizeof head);
    memset(nxt,0,sizeof nxt);
    memset(node,0,sizeof node);
    for (i=1;i<=m;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);
    }
}

void Dij(){
    memset(d,10,sizeof d);d[1]=0;
    priority_queue<pair<int,int> > heap;
    heap.push(make_pair(-d[1],1));
    while (1){
        for (;!heap.empty() && -d[heap.top().second]!=heap.top().first;heap.pop());
        //对于每个d[i]，只会被更新的越来越小，所以如果当前的堆顶的原本的d[]不等于现在的d[]，说明肯定已经被更新过，不是最优的 
        
		if (heap.empty()) break;	//一开始没加这句话 错的非常惨。。。。 调了一整天。。。。 
        
        int now=heap.top().second;
        heap.pop();
        
        for (i=head[now];i;i=nxt[i]){
            j=node[i];
            if (d[j]>d[now]+data[i]){
                d[j]=d[now]+data[i];
                heap.push(make_pair(-d[j],j));
            }
        }
    }
}

int main(){
    for (scanf("%d%d",&n,&m);n && m;scanf("%d%d",&n,&m)){
        init();
        Dij();
        printf("%d\n",d[n]);
    }
    return 0;
}